package com.qf.controller;

import com.baomidou.mybatisplus.core.metadata.IPage;
import com.qf.annotation.RequiresPermissionsDesc;
import com.qf.pojo.DtsRole;
import com.qf.service.DtsPermissionService;
import com.qf.service.DtsRoleService;
import com.qf.util.Permission;
import com.qf.util.PermissionUtil;
import com.qf.util.ResponseUtil;
import com.qf.vo.PermVo;
import org.apache.shiro.authz.annotation.RequiresPermissions;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.context.ApplicationContext;
import org.springframework.web.bind.annotation.*;

import java.util.HashMap;
import java.util.List;
import java.util.Map;
import java.util.Set;
import java.util.stream.Collectors;

/**
 * @author Administrator
 * @version v1.0
 * @date 2022/5/18
 */
@RestController
@RequestMapping("/admin/role")
public class DtsAdminRoleController {
    @Autowired
    DtsRoleService roleService;

    @Autowired
    DtsPermissionService permissionService;
    @Autowired
    ApplicationContext context;

    @PostMapping("/permissions")
    public Object grantPermission(@RequestBody Map<String,Object> map){
        Integer roleId =(Integer) map.get("roleId");
        List<String> permissions =(List<String>) map.get("permissions");
        permissionService.update(roleId,permissions);
        return  ResponseUtil.ok();
    }

    @GetMapping("/permissions")
    public Object permissions(Integer roleId){

        List<Permission> permissions = PermissionUtil.listPermission(context, "com.qf.controller");
        Set<String> permissionStrs = PermissionUtil.listPermissionString(permissions);
        List<PermVo> systemPermissions = PermissionUtil.listPermVo(permissions);
        Map<String,Object>  data = new HashMap<>();
        data.put("systemPermissions", systemPermissions);


        if(roleService.isSuperAdmin(roleId)){
            data.put("assignedPermissions", permissionStrs);
        }else{
            Set<String> assignedPermissions = permissionService.findPermByIds(new Integer[]{roleId});
            data.put("assignedPermissions", assignedPermissions);
        }
        return ResponseUtil.ok(data);
    }

    @RequiresPermissions("admin:role:create")
    @RequiresPermissionsDesc(menu = { "系统管理", "角色管理" }, button = "角色添加")
    @GetMapping("/create")
    public Object create(@RequestBody DtsRole role){
        DtsRole role1 = roleService.insert(role);

        return ResponseUtil.ok(role1);
    }
    @RequiresPermissions("admin:role:delete")
    @RequiresPermissionsDesc(menu = { "系统管理", "角色管理" }, button = "角色删除")
    @PostMapping("/delete")
    public Object delete(@RequestBody DtsRole role){
        boolean flag = roleService.delete(role);
        if(flag){
            return ResponseUtil.ok();
        }else{
            return ResponseUtil.fail(-1,"该角色不能删除");
        }
    }

    @RequiresPermissions("admin:role:update")
    @RequiresPermissionsDesc(menu = { "系统管理", "角色管理" }, button = "角色编辑")
    @PostMapping("/update")
    public Object update(@RequestBody DtsRole role){
        roleService.update(role);
        return ResponseUtil.ok();
    }
    @RequiresPermissions("admin:role:list")
    @RequiresPermissionsDesc(menu = { "系统管理", "角色管理" }, button = "角色查询")
    @GetMapping("/list")
    public Object list(
            @RequestParam()Integer page,
            @RequestParam()Integer limit,
            String sort,String order,String rolename
    ){
        IPage<DtsRole> iPage = roleService.queryPage(page,limit,sort,order,rolename);
        Map<String,Object> data = new HashMap<>();
        data.put("total", iPage.getTotal());
        data.put("items", iPage.getRecords());

        return ResponseUtil.ok(data);
    }
    @RequestMapping("/options")
    public Object options(){
        List<DtsRole> roleList = roleService.findAll();
        List<Map<String,Object>> data =roleList.stream().map(role->{
            Map<String,Object> map = new HashMap<>();
            map.put("value",role.getId());
            map.put("label", role.getName());
            return map;
        }).collect(Collectors.toList());
        return ResponseUtil.ok(data);
    }
}
